from argparse import Namespace
import torch

config = Namespace(
    project_name='steel',
    AVAIL_GPUS=min(1, torch.cuda.device_count()),
    modelPATH="modelPATH/last.ckpt",
    data_dir ="../../../../data/steel/新分类样本/",

)

